package com.changgou.file.controller;

import com.changgou.entity.Result;
import com.changgou.entity.StatusCode;
import com.changgou.file.util.FastDFSClient;
import com.changgou.file.util.FastDFSFile;
import org.apache.commons.lang.StringUtils;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;

@RestController
@RequestMapping("/file")
public class FileController {

    @PostMapping("/upload")
    public Result uploadFile(MultipartFile file) {
        //判断文件是否存在
        if (null != file) {
            //获取文件完整名称
            String originalFilename = file.getOriginalFilename();

            if (StringUtils.isBlank(originalFilename)) {
                throw new RuntimeException("文件内容为空");
            } else {
                //获取文件扩展名
                String extname = originalFilename.substring(originalFilename.lastIndexOf(".") + 1);
                try {
                    //获取文件内容
                    byte[] content = file.getBytes();
                    //创建文件上传的封装实体类
                    FastDFSFile fastDFSFile = new FastDFSFile(originalFilename, content, extname);
                    //基于工具类进行上传,接收返回参数 String[]
                    String[] uploadResult = FastDFSClient.upload(fastDFSFile);
                    //封装返回结果
                    String url = FastDFSClient.getTrackerUrl() + uploadResult[0] + "/" + uploadResult[1];
                    return new Result(true, StatusCode.OK, "文件上传成功", url);
                } catch (IOException e) {
                    e.printStackTrace();
                    return new Result(false, StatusCode.ERROR, "文件上传失败");
                }
            }
        } else {
            throw new RuntimeException("文件不存在");
        }
    }

}
